背景與起源
Hugging Face 成立於 2016 年,最初是一家開發聊天機器人的公司,但隨著人工智慧快速發展,它逐漸轉型為自然語言處理與人工智慧AI的核心開源生態系。如今,Hugging Face不僅提供上萬個開源模型與資料集,也成為 AI 社群的重要平台,支持研究人員、開發者與企業快速使用和分享最先進的AI技術。
主要功能
(1) Transformers 套件
Hugging Face 的旗艦產品,支援多種深度學習框架(PyTorch、TensorFlow)。
提供簡單 API,用戶可快速載入預訓練模型並進行推論或微調。
首先安裝Transformers套件
pip install transformers datasets
安裝完成就可以導入裡面的套件來進行模型訓練了
from transformers import pipeline
# 情感分析(指定模型名稱)
classifier = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
result = classifier("I love learning about Hugging Face!")
print(result)
# 文字生成(GPT-2)
generator = pipeline("text-generation", model="gpt2")
text = generator("Hugging Face is a great tool because", max_length=50, num_return_sequences=1)
print(text[0]['generated_text'])